Type: concept
Confidence: 0.80
Created: 2026-04-16
Updated: 2026-04-16
Tags: 推荐系统训练策略RNNGPU并行负采样

Session-Parallel Mini-Batch

概述

GRU4Rec 提出的 RNN 训练策略,将多个会话并排放置逐步推进,充分利用 GPU 并行计算能力,避免长短不一会话带来的 padding 浪费,同时天然提供负采样机制。

关键内容

  1. 核心机制:将多个会话并排放置,每个 mini-batch 包含来自不同会话的同一"步骤"的事件;每步取各会话的当前事件作为输入、下一个事件作为目标;当某会话结束时,替换为新会话并重置对应位置的 GRU 隐藏状态;权重在每步都更新,而非等待整个会话完成。

  2. 与传统 BPTT 的对比:传统方式将一个会话作为一个样本,对会话内所有时间步做反向传播(BPTT)。但会话长度差异巨大(2 次到几十次点击不等),导致效率低下且实现复杂。Session-Parallel Mini-Batch 提供了一种优雅的折中方案。

  3. 关键优势

  4. 计算效率:充分利用 GPU 并行计算能力,避免长短不一会话带来的 padding 浪费
  5. 隐藏状态管理:每个会话独立维护自己的 GRU 隐藏状态,会话结束后状态归零
  6. 负采样便利:mini-batch 中其他会话的目标物品天然作为当前会话的负样本,无需额外采样操作
  7. 流行度采样:热门物品更容易被采到作为负样本,与 popularity-based sampling 一致

  8. 后续影响:这一训练范式被广泛采用,成为序列推荐模型训练的标准方法之一。后续的 NARM、SASRecBERT4Rec 等模型均采用了类似的批处理策略。

  9. 工程意义:体现了 GRU4Rec 论文务实的工程思维——将计算效率作为设计约束,而非仅追求理论优美。这种理论与实践并重的态度是论文被工业界广泛采纳的重要原因。

来源

相关